1055A - Metro - CodeForces Solution


graphs *900

Please click on ads to support us..

Python Code:

n, m = map(int, input().split())
a = [int(x) for x in input().split()]
b = [int(x) for x in input().split()]
def func(m, n):
    for j in range(m, n):
        if a[j] == 1 and b[j] == 1:
            return "YES"
    return "NO"

if a[0] == 1:
    if a[m - 1] == 1:
        print("YES")
    else:
        if b[m - 1] == 1:
            print(func(m, n))
        else:
            print("NO")
else:
    print("NO")

C++ Code:

  #include<bits/stdc++.h>
  #define ll long long

  using namespace std;

  int main()
  {
      ios::sync_with_stdio(0);
      cout.tie(0);
      cin.tie(0);

      int n, x; cin >> n >> x;
      int track1[n];
      int track2[n];
      string result = "NO";
      for(int i = 0; i < n; i++)
          cin >> track1[i];
      for(int i = 0; i < n; i++)
          cin >> track2[i];
      if(track1[0] == 1)
      {
          if(track1[x-1] == 1)
            result = "YES";
          else if(track2[x-1] == 1)
          {
              for(int i = x; i < n; i++)
              {
                  if(track1[i] == 1 && track2[i] == 1)
                  {
                    result = "YES";
                    break;
                  }
              }
          }
      }
      cout << result << endl;

      return 0;
  }



Comments

Submit
0 Comments
More Questions

873D - Merge Sort
1251A - Broken Keyboard
463B - Caisa and Pylons
584A - Olesya and Rodion
799A - Carrot Cakes
1569B - Chess Tournament
1047B - Cover Points
1381B - Unmerge
1256A - Payment Without Change
908B - New Year and Buggy Bot
979A - Pizza Pizza Pizza
731A - Night at the Museum
742A - Arpa’s hard exam and Mehrdad’s naive cheat
1492A - Three swimmers
1360E - Polygon
1517D - Explorer Space
1230B - Ania and Minimizing
1201A - Important Exam
676A - Nicholas and Permutation
431A - Black Square
474B - Worms
987B - High School Become Human
1223A - CME
1658B - Marin and Anti-coprime Permutation
14B - Young Photographer
143A - Help Vasilisa the Wise 2
320A - Magic Numbers
1658A - Marin and Photoshoot
514A - Chewbaсca and Number
382A - Ksenia and Pan Scales